#!/usr/bin/python
# -*-coding:utf-8-*-
import os
import pandas as pd
from datetime import datetime

# from zg02_factor_lib.dir_info import logs_data_dir
from dir_info import logs_dir as logs_data_dir

import logging
from logging import handlers

class Logger(object):
    level_relations = {
        'debug':logging.DEBUG,
        'info':logging.INFO,
        'warning':logging.WARNING,
        'error':logging.ERROR,
        'crit':logging.CRITICAL
    }#日志级别关系映射

    def __init__(self,filename,level='info',when='D',backCount=3,fmt='%(asctime)s - %(levelname)s: %(message)s'):
        self.logger = logging.getLogger(filename)
        format_str = logging.Formatter(fmt)#设置日志格式
        self.logger.setLevel(self.level_relations.get(level))#设置日志级别
        sh = logging.StreamHandler()#往屏幕上输出
        sh.setFormatter(format_str) #设置屏幕上显示的格式
        th = handlers.TimedRotatingFileHandler(filename=filename,when=when,backupCount=backCount,encoding='utf-8')#往文件里写入#指定间隔时间自动生成文件的处理器
        #实例化TimedRotatingFileHandler
        #interval是时间间隔，backupCount是备份文件的个数，如果超过这个个数，就会自动删除，when是间隔的时间单位，单位有以下几种：
        # S 秒
        # M 分
        # H 小时、
        # D 天、
        # W 每星期（interval==0时代表星期一）
        # midnight 每天凌晨
        th.setFormatter(format_str)#设置文件里写入的格式
        self.logger.addHandler(th)

# # TODO - 设置log
# factor_update_date = datetime.now() - pd.Timedelta(days=1)
# factor_update_date = factor_update_date.strftime('%Y%m%d')
#
# # logger_data_dir = logs_data_dir
# # if not os.path.exists(logger_data_dir):
# #     os.makedirs(logger_data_dir)
#
# factor_update_logger = Logger(os.path.join(logs_data_dir,
#                               'zg02_factor_update_test_log_%s.log' % factor_update_date), level='info')
#
